package io.renren.modules.doc.controller;

import io.renren.common.utils.PageUtils;
import io.renren.common.utils.R;
import io.renren.modules.doc.entity.PromptTemplateEntity;
import io.renren.modules.doc.service.PromptTemplateService;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.List;
import java.util.Map;

/**
 * 提示词模板管理
 */
@RestController
@RequestMapping("/doc/prompt")
public class PromptTemplateController {
    @Autowired
    private PromptTemplateService promptTemplateService;

    /**
     * 列表
     */
    @GetMapping("/list")
    @RequiresPermissions("doc:prompt:list")
    public R list(@RequestParam Map<String, Object> params){
        PageUtils page = promptTemplateService.queryPage(params);
        return R.ok().put("page", page);
    }

    /**
     * 根据分类获取模板
     */
    @GetMapping("/category/{category}")
    @RequiresPermissions("doc:prompt:list")
    public R getByCategory(@PathVariable("category") String category){
        List<PromptTemplateEntity> templates = promptTemplateService.getTemplatesByCategory(category);
        return R.ok().put("templates", templates);
    }

    /**
     * 根据标签搜索模板
     */
    @GetMapping("/search")
    @RequiresPermissions("doc:prompt:list")
    public R search(@RequestParam("tags") String[] tags){
        List<PromptTemplateEntity> templates = promptTemplateService.searchByTags(tags);
        return R.ok().put("templates", templates);
    }

    /**
     * 获取热门模板
     */
    @GetMapping("/popular")
    @RequiresPermissions("doc:prompt:list")
    public R popular(@RequestParam(defaultValue = "10") int limit){
        List<PromptTemplateEntity> templates = promptTemplateService.getPopularTemplates(limit);
        return R.ok().put("templates", templates);
    }

    /**
     * 增加使用次数
     */
    @PostMapping("/use/{id}")
    @RequiresPermissions("doc:prompt:update")
    public R incrementUsage(@PathVariable("id") Long id){
        promptTemplateService.incrementUsageCount(id);
        return R.ok();
    }

    /**
     * 信息
     */
    @GetMapping("/info/{id}")
    @RequiresPermissions("doc:prompt:info")
    public R info(@PathVariable("id") Long id){
        PromptTemplateEntity promptTemplate = promptTemplateService.getById(id);
        return R.ok().put("promptTemplate", promptTemplate);
    }

    /**
     * 保存
     */
    @PostMapping("/save")
    @RequiresPermissions("doc:prompt:save")
    public R save(@RequestBody PromptTemplateEntity template){
        promptTemplateService.saveTemplate(template);
        return R.ok();
    }

    /**
     * 修改
     */
    @PostMapping("/update")
    @RequiresPermissions("doc:prompt:update")
    public R update(@RequestBody PromptTemplateEntity template){
        promptTemplateService.updateTemplate(template);
        return R.ok();
    }

    /**
     * 删除
     */
    @PostMapping("/delete")
    @RequiresPermissions("doc:prompt:delete")
    public R delete(@RequestBody Long[] ids){
        promptTemplateService.deleteTemplate(ids);
        return R.ok();
    }
}
